#!/usr/bin/env bash

set -e  # 出错时立即停止脚本

# 可选：安装 pipx（如果尚未安装）
if ! command -v pipx &> /dev/null; then
    echo "Installing pipx..."
    python3 -m pip install --user pipx
    python3 -m pipx ensurepath
fi

# 可选：安装 nvitop-exporter（如果尚未安装）
if ! command -v nvitop-exporter &> /dev/null; then
    echo "Installing nvitop-exporter..."
    pipx install nvitop-exporter
fi

# 确保 pipx 的路径生效（适用于新 shell）
export PATH="$PATH:$HOME/.local/bin"

# 启动服务
echo "Starting nvitop-exporter on 0.0.0.0:5050..."
exec nvitop-exporter --bind-address 0.0.0.0 --port 5050
echo "nvitop-exporter start success!"
